package com.yzqp.generator.util.generator.filter.map.add;

import com.yzqp.generator.bean.TableColumnInfo;
import com.yzqp.generator.util.HumpNameUtil;
import com.yzqp.generator.util.generator.filter.map.GeneratorMapFilter;

import java.util.List;
import java.util.Map;

/**
 * 生成器的 map 的过滤器
 * <p>
 * 添加基础参数
 */
public class AddBaseTempleParamMapFilter implements GeneratorMapFilter {

    @Override
    public Map<String, Object> filter(Map<String, Object> map, List<TableColumnInfo> tableColumnInfoList) {
        map.put("columnList", tableColumnInfoList);
        // 第一列
        TableColumnInfo columnItem = tableColumnInfoList.get(0);
        // 表注释
        map.put("tableComment", columnItem.getTableComment());
        // 类注释
        map.put("javaClassComment", columnItem.getTableComment());
        // 表名
        map.put("tableName", columnItem.getTableName());
        // controller路径
        // 基础类名
        map.put("controllerPath", HumpNameUtil.getControllerPath(columnItem.getTableName()));
        return map;
    }
}
